#coding:utf-8
import time
from io import BytesIO
from PIL import Image
from selenium import webdriver
from selenium.webdriver import ActionChains
from selenium.webdriver.support.wait import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By

'''
    作用：进行市面上滑块验证码的自动验证
'''

class CrackGeetest():
    def __init__(self):
        self.url = 'https://auth.geetest.com/register'
        self.browser = webdriver.Chrome()
        self.wait = WebDriverWait(self.browser, 20)
        self.email = 'huazi@163.com'


    def open(self):
        '''
            打开网页并输入表单数据（邮箱）
        '''
        self.browser.get(self.url)
        email = self.wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'ivu-input')))
        email.send_keys(self.email)

    def get_geetest_button(self):
        '''
            作用：获取初始验证按钮
            return：按钮对象
        '''
        button = self.wait.until(EC.element_to_be_clickable((By.CSS_SELECTOR, '.geetest_radar_tip')))
        return button

    def get_code_img(self, name='captcha.png'):
        '''
            作用：获取验证码图片
            return：图片对象
        '''
        #获取验证码位置
        img = self.wait.until(EC.presence_of_element_located((By.CLASS_NAME, 'geetest_canvas_img')))
        time.sleep(2)
        location = img.location
        size = img.size
        top, bottom, left, right = location['y'], location['y'] + size['height'], location['x'], location['x'] + size['width']
        print('验证码位置', top, bottom, left, right)

        #调用截图接口，获取截图图片
        screenshot = self.get_screenshot()
        captcha = screenshot.crop((left, top, right, bottom))
        captcha.save(name)
        return captcha

    def get_screenshot(self):
        """
        获取网页截图
        :return: 截图对象
        """
        screenshot = self.browser.get_screenshot_as_png()
        screenshot = Image.open(BytesIO(screenshot))
        return screenshot

    def get_slider(self):
        '''
            作用：获取滑块
            return：滑块对象
        '''
        slider = self.wait.until(EC.element_to_be_clickable((By.CLASS_NAME, 'geetest_slider_button')))
        return slider

    def get_gap(self, image1, image2):
        '''
            作用：获取缺口偏移量
            :param image1： 不带缺口的图片
            :param image2: 带缺口图片
            :return 缺口左侧偏移量
        '''    
        left = 60
        #从左侧开始验证图片：i为图片的总宽，j为图片总高。通过两个参数不断比较图片在这个像素内RGB值是否有变化
        for i in range(left, image1.size[0]):
            for j in range(image1.size[1]):
                if not self.is_pixel_equal(image1, image2, i, j):
                    left = i
                    return left
        return left

    def is_pixel_equal(self, image1, image2, x, y):
        '''
            作用：判断两个图片像素是否相同
            :param image1: 图片1
            :param image2: 图片2
            :param x: 位置x
            :param y: 位置y
            :return: 像素是否相同
        '''
        #去两个图片的像素点
        pixel1 = image1.load()[x, y]
        pixel2 = image2.load()[x, y]
        #像素的误差值（阀值）
        threshold = 60
        #判断每个位置的像素点的RGB值3个参数是否在误差值内：在则表示图片相同
        if abs(pixel1[0] - pixel2[0]) < threshold and abs(pixel1[1] - pixel2[1]) < threshold and abs(pixel1[2] - pixel2[2]) < threshold:
            return True
        else:
            return False


    def get_track(self, distance):
        '''
            作用：根据偏移量获取移动轨迹
            :param distance: 偏移量
            :return: 移动轨迹
        '''
        #移动轨迹
        track = []
        #当前位移
        current = 0
        #减速阀值
        mid = distance * 4/5
        #计算间隔
        t = 0.2
        #初速度
        v = 0

        while current < distance:
            if current < mid:
                #加速度为正2
                a = 2
            else:
                #加速度为-3
                a = -3
            #初始速度v0
            v0 = v
            #当前速度v = v0 + at
            v = v0 + a * t
            #移动距离 x = v0t + 1/2 * a * t^2
            move = v0 * t + 1/2 * a * t * t
            #当前位移
            current += move
            #加入轨迹
            track.append(round(move))
        return track

    
    def move_to_gap(self, slider, track):
        '''
            作用：滑动滑块到缺口处
            :param slider:滑块
            :param track:轨迹
        '''
        ActionChains(self.browser).click_and_hold(slider).perform()
        for x in track:
            ActionChains(self.browser).move_by_offset(xoffset=x, yoffset=0).perform()
        time.sleep(0.5)
        ActionChains(self.browser).release().perform()

    def crack(self):
        #打开页面
        self.open()
        #获取初始按钮，并点击
        button = self.get_geetest_button()
        button.click()
        #获取初始图片
        initImg = self.get_code_img()
        #点击验证滑块
        slider = self.get_slider()
        slider.click()
        #获取带有验证缺口的图片
        codeImg = self.get_code_img()
        #获取缺口偏移量
        gap = self.get_gap(initImg, codeImg)
        print('缺口位置', gap)
        # 减去缺口位移
        gap -= 6
        #获取移动轨迹
        track = self.get_track(gap)
        print('滑动轨迹', track)
        #拖动滑块
        self.move_to_gap(slider, track)

        success = self.wait.until(EC.text_to_be_present_in_element((By.CLASS_NAME, 'geetest_success_radar_tip_content'), '验证成功'))
        print(success,'验证成功')

if __name__ == '__main__':
    geetest = CrackGeetest()
    geetest.crack()



